import matplotlib.pyplot as plt
import numpy as np

np.random.seed(19680801)


fig, ax = plt.subplots()

x1=[0.375745,
0.355075,
0.327195,
0.311135,
0.308505,
0.303445,
0.29868,
0.299665,
0.296295,
0.29852,
0.304445,
0.30292,
0.30626,
0.30825,
0.307095,
0.306965,
0.307815,
0.3075,
0.3085,
0.30842]

y1 = [73.40608,
77.66204,
79.2791,
80.20833,
80.376965,
80.41336,
81.187175,
81.50463,
81.5443,
81.66004,
81.97751,
82.506605,
82.410715,
82.71494,
83.08532,
83.13162,
83.34656,
83.445765,
83.458995,
83.488755,
]


x2 = [
    0.9256,
0.8881,
0.8862,
0.89,
0.876,
0.8441,
0.8675,
0.8613,
0.8597,
0.8512,
0.8572,
0.8595,
0.8579,
0.8562,
0.8599,
0.8541,
0.8539,
0.8503,
0.8536,
0.8531
]

y2 =[
    71.4286,
77.2487,
77.1164,
80.1587,
79.6958,
78.7698,
80.4894,
80.8201,
83.0026,
81.4153,
80.6878,
82.8704,
82.3413,
83.2011,
84.3254,
83.4656,
83.5979,
83.5979,
83.2011,
83.9947,
]

scale=np.array([
    1,
1,
1,
1,
1,
2,
2,
2,
2,
2,
2.333333333,
2.484848485,
2.636363636,
2.787878788,
2.939393939,
3.090909091,
3.242424242,
3.393939394,
3.545454545,
3.696969697
])

x1,y1,x2,y2 = np.array(x1),np.array(y1),np.array(x2),np.array(y2)
x2 = x2-0.4
y2=y2-2
scale = 100*scale

colors = ['tab:blue', 'tab:orange']

labels = ['ViT','OVIT']

ax.scatter(x1, y1, c=colors[0], s=scale, label=labels[0],
            alpha=0.3, edgecolors='none')



ax.scatter(x2, y2, c=colors[1], s=scale, label=labels[1],
            alpha=0.3, edgecolors='none')

ax.legend()
ax.grid(True)

plt.savefig('Main_results/2_fig2/1_trajectory.png')
# plt.show()